# -*- coding: utf-8 -*-
import csv


'''
博文信息服务类
'''


class ArticleInfoDao:

    def __init__(self):
        pass

    '''
    记录信息到CSV文件里
    '''

    @classmethod
    def create_csv(cls, path):
        with open(path, "w", newline="") as datacsv:
            # dialect为打开csv文件的方式，默认是excel，delimiter="\t"参数指写入的时候的分隔符
            csvwriter = csv.writer(datacsv, dialect=("excel"))
            # csv文件插入一行数据，把下面列表中的每一项放入一个单元格（可以用循环插入多行）
            csvwriter.writerow(["博文ID",
                                "标题",
                                "标签",
                                "分类",
                                "作者ID",
                                "博客ID",
                                "是否原创",
                                "点击量",
                                "评论数",
                                "发布日期",
                                "访问链接"
                                ])

    '''
    记录信息到CSV文件里
    博文ID， 标题，标签， 分类， 作者ID, 博客ID, 是否原创， 点击量， 评论数， 发布日期， 访问链接 
    '''
    @classmethod
    def add_item_to_csv(cls, path, item):
        with open(path, "a", newline="") as datacsv:
            # dialect为打开csv文件的方式，默认是excel，delimiter="\t"参数指写入的时候的分隔符
            csvwriter = csv.writer(datacsv, dialect=("excel"))
            # csv文件插入一行数据，把下面列表中的每一项放入一个单元格（可以用循环插入多行）
            csvwriter.writerow([item["article_id"],
                                item["title"],
                                item["tag_name"],
                                item["class_name"],
                                item["user_id"],
                                item["blog_id"],
                                item["original"],
                                item["click_count"],
                                item["comment_count"],
                                item["publish_date"],
                                item["url"]
                               ])

